package com.lagou.module4hw.hw1;

import java.io.*;
import java.util.InputMismatchException;
import java.util.LinkedList;
import java.util.Scanner;

/**
 * 学生管理系统入口
 */
public class Main {

    //声明LinkedList数组，里面存放学生类信息
    public static LinkedList<Student> stus = null;

    /**
     * 将学生信息写入文件中
     * @param stus 学生集合
     */
    public static void writeStuInfo(LinkedList<Student> stus){
        //1.创建输出对象流
        ObjectOutputStream oos = null;
        try {
            oos = new ObjectOutputStream(new FileOutputStream("./stus.info"));
            //3.将学生信息序列化写入文件中
            oos.writeObject(stus);
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
               if(oos != null) oos.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    /**
     * 读取学生信息
     * @return stus 学生的信息
     */
    public static LinkedList<Student> readStuInfo(){
        //1.从文件中读取对象
        ObjectInputStream ois = null;
        try {
            //先判断文件是否存在，存在则读取，不存在返回空队列
            File file = new File("./stus.info");
            if ( !file.exists()) {
                return new LinkedList<Student>();
            }

            ois = new ObjectInputStream( new FileInputStream("./stus.info"));
            try {
                LinkedList<Student> stu = (LinkedList<Student>) ois.readObject();
                return stu;
            } catch (ClassNotFoundException e) {
                e.printStackTrace();
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (ois != null) {
                try {
                    ois.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        return new LinkedList<Student>();
    }

    public static void main(String[] args) {
        //声明变量存储用户选择
        int choose;
        Scanner scanner =  new Scanner(System.in);

        //读取文件中的信息
        Main.stus = readStuInfo();

        boolean flag = true;
        while(flag) {
            Menu.show();
            try {

                System.out.println("请输入您的选择:");
                choose = Integer.parseInt(scanner.next());
                switch (choose) {
                    case 1://添加学习信息
                        AddStudent.setStus(Main.stus);
                        while (true) {
                            Menu.showAdd();
                            String str = scanner.next();
                            if (AddStudent.add(str)) break;
                        }
                        break;
                    case 2://修改学生信息
                        ModifyStudent.setStus(Main.stus);
                        while (true) {
                            Menu.showModify();
                            String str = scanner.next();
                            if (ModifyStudent.modifyStu(str)) break;
                        }
                        break;
                    case 3: //删除学生信息
                        DelStudent.setStus(Main.stus);
                        while (true) {
                            Menu.showDel();
                            String str = scanner.next();
                            if (DelStudent.delStu(str)) break;
                        }
                        break;
                    case 4: //查看学生信息
                        ShowStudent.setStus(Main.stus);
                        ShowStudent.showStus();
                        System.out.println("输入任意键回到主界面");
                        String s = scanner.next();
                        break;
                    case 5://退出系统
                        Main.writeStuInfo(Main.stus);
                        flag = false;
                        break;
                    default:
                        System.out.println("输入的选择不存在，请重新输入");
                        break;
                }
            }catch (InputMismatchException e){
                System.out.println("输入错误，请输入数字\t1\t2\t3\t4\t5");
            }catch (NumberFormatException e){
                System.out.println("输入错误，请输入数字\t1\t2\t3\t4\t5");
            }
        }
    }
}
